package com.mysql.injector.injectror;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
import com.mysql.injector.method.InsertIgnore;
import com.mysql.injector.method.InsertIgnoreBatchSomeColumn;
import com.mysql.injector.method.InsertOnDuplicateKeyUpdate;
import com.mysql.injector.method.InsertOnDuplicateKeyUpdateBatchSomeColumn;
import com.mysql.injector.method.ReplaceIntoBatchSomeColumn;
import java.util.List;

/**
 * 自定义sql 注入器
 */
public class MySqlInjector extends DefaultSqlInjector {

  @Override
  public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {

    List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);

    // 增加自定义方法
    // 批量新增数据,自选字段（mybatis plus自带）
    methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));

    // 批量替换新增数据,自选字段
    methodList.add(new ReplaceIntoBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));

    // 批量新增数据（重复存在忽略插入）,自选字段
    methodList.add(new InsertIgnoreBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));

    // 批量新增数据（重复存在更新数据）,自选字段
    methodList.add(new InsertOnDuplicateKeyUpdateBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));

    // 单条数据插入（重复存在忽略插入）
    methodList.add(new InsertIgnore());

    // 单条数据插入（重复存在更新数据）
    methodList.add(new InsertOnDuplicateKeyUpdate());

    return methodList;
  }

}
